'''
Sample E
One possible solution to flowchart E
'''
from codex import *
from random import randrange


if buttons.was_pressed(BTN_A):
    red = randrange(256)
    green = randrange(256)
    blue = randrange(256)
    color = (red, green, blue)
    pixels.set(0, color)
    pixels.set(1, color)
    pixels.set(2, color)
    pixels.set(3, color)


if buttons.was_pressed(BTN_B):
    pixels.set(0, WHITE)
    pixels.set(1, WHITE)
    pixels.set(2, WHITE)
    pixels.set(3, WHITE)